import os
from dotenv import load_dotenv

# 设置默认环境
default_env = 'dev'  # 可以改为 'local', 'dev', 或 'prod'
ENV = os.getenv('ENV', default_env)

# 获取当前文件所在目录
current_dir = os.path.dirname(os.path.abspath(__file__))
# 获取项目根目录
root_dir = os.path.dirname(current_dir)

# 根据环境选择配置文件
if ENV == 'local':
    env_file = '.env.local'
elif ENV == 'dev':
    env_file = '.env.dev'
elif ENV == 'prod':
    env_file = '.env'
else:
    env_file = '.env.local'

# 构造配置文件的绝对路径
env_path = os.path.join(root_dir, env_file)

print("开始加载配置...")
print(f"当前工作目录: {os.getcwd()}")
print(f"尝试加载配置文件: {env_path}")
print(f"文件是否存在: {os.path.exists(env_path)}")

# 加载配置文件
load_dotenv(dotenv_path=env_path, verbose=True)

print("\n环境变量加载结果:")
print(f"MYSQL_HOST = {os.getenv('MYSQL_HOST')}")
print(f"OSS_BUCKET_NAME = {os.getenv('OSS_BUCKET_NAME')}")

# 数据库配置
MYSQL_CONFIG = {
    'host': os.getenv('MYSQL_HOST'),
    'port': int(os.getenv('MYSQL_PORT')),
    'user': os.getenv('MYSQL_USER'),
    'password': os.getenv('MYSQL_PASSWORD'),
    'database': os.getenv('MYSQL_DATABASE')
}

# OSS配置
OSS_CONFIG = {
    'access_key': os.getenv('OSS_ACCESS_KEY'),
    'access_secret': os.getenv('OSS_ACCESS_SECRET'),
    'bucket_name': os.getenv('OSS_BUCKET_NAME'),
    'endpoint': os.getenv('OSS_ENDPOINT')
} 